package com.example.mqltest.LogConfig;

import com.google.gson.Gson;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.*;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import javax.servlet.http.HttpServletRequest;

/**
 * @author zhanghong
 * @date 2021/4/9  0009 19:16
 * @description 控制台打印日志
 */
@Component
@Aspect
public class WebLogAspect {
    private final static Logger logger = LoggerFactory.getLogger( com.example.mqltest.LogConfig.WebLogAspect.class );

    /**
     * 以 controller 包下定义的所有请求为切入点
     *
     */
    @Pointcut("execution(public * com.example.mqltest.controler.*.*(..))")
    public void webLog() {
    }

    /**
     *     * 在切点之前织入
     *     * @param joinPoint
     *     * @throws Throwable
     *    
     */
    @Before("webLog()")
    public void doBefore(JoinPoint joinPoint) throws Throwable {
        // 开始打印请求日志
        ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
        HttpServletRequest request = attributes.getRequest();
// 打印请求相关参数
        logger.info( "========================================== Start ==========================================" );
// 打印请求 url
        logger.info( "URL            : {}", request.getRequestURL().toString() );
// 打印 Http method
        logger.info( "HTTP Method    : {}", request.getMethod() );
// 打印调用 controller 的全路径以及执行方法
        logger.info( "Class Method   : {}.{}", joinPoint.getSignature().getDeclaringTypeName(), joinPoint.getSignature().getName() );
        // 打印请求的 IP
        logger.info( "IP             : {}", request.getRemoteAddr() );
        // 打印请求入参
        logger.info( "Request Args   : {}", new Gson().toJson( joinPoint.getArgs() ) );
    }

    /**
     *     * 在切点之后织入
     *     * @throws Throwable
     *    
     */
    @After("webLog()")
    public void doAfter() throws Throwable {
// 每个请求之间空一行
        logger.info( "" );
    }

    /**
     *     * 环绕
     *     * @param proceedingJoinPoint
     *     * @return
     *     * @throws Throwable
     *    
     */
    @Around("webLog()")
    public Object doAround(ProceedingJoinPoint proceedingJoinPoint) throws Throwable {
        long startTime = System.currentTimeMillis();
        Object result = proceedingJoinPoint.proceed();
// 打印出参数
        logger.info( "Response Args  : {}", new Gson().toJson( result ) );
  // 执行耗时
        logger.info( "Time-Consuming : {} ms", System.currentTimeMillis() - startTime );
        logger.info( "=========================================== End ===========================================" );
        return result;
    }

}
